// Copyright 2024 Ant Group Co., Ltd.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package mods

import (
	"context"
	"fmt"

	"github.com/secretflow/kuscia/pkg/diagnose/app/netstat"
)

type Mod interface {
	Run(ctx context.Context) error
}

type DiagnoseConfig struct {
	Command     string
	Name        string
	Source      string
	Destination string
	CRDType     string
	ReportFile  string
	*netstat.NetworkParam
}

func (c *DiagnoseConfig) String() string {
	return fmt.Sprintf(`--Command: %s
--Source: %s
--Destination: %s
--CRD: %s
--ReportFile: %s
--TestSpeed: %v, Threshold: %v
--TestRTT: %v, Threshold: %v
--TestProxyTimeout: %v, Threshold: %v
--TestProxyBuffer: %v
--TestRequestBodySize: %v, Threshold: %v
--BidrectionMode: %v`, c.Command, c.Source, c.Destination, c.CRDType, c.ReportFile, c.Speed, c.SpeedThres, c.RTT, c.RTTTres, c.ProxyTimeout, c.ProxyTimeoutThres, c.ProxyBuffer, c.Size, c.SizeThres, c.Bidirection)
}
